yolov5目标检测用IOU提高预测准确[yolo

您所在的位置:网站首页 yolov5 iou阈值 yolov5目标检测用IOU提高预测准确[yolo

yolov5目标检测用IOU提高预测准确[yolo

2024-07-03 06:18| 来源: 网络整理| 查看: 265

yolo预测出了多个边界框,我们选择哪一个呢?

不可以选择最大概率的框。以狗狗举例,概率分别为0.9;0.7;0.6,若画面中还有其他的狗狗的概率框为0.91,那么就会漏掉所见到的可爱的金毛。会漏检。

采用IOU交并比的NMS(非极大值抑制)解决(Intersection over Union)

yolo会把最大概率的边界框先拉出来,和其他框进行对比,进行iou的计算

当iou的数值大于给定的阈值如0.5,则保留;如果没有大于0.5的框就丢弃掉

当剩下的iou有好几个都是大于0.5的,yolo就会保留最大的作为边界框。这应该就是NMS的过程。

如果画面中会有其他的猫或者狗

不同的猫与猫 不同的狗与狗之间是不会有框的交集的,因此在做iou计算时彼此并没有影响。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3